package com.tensai.study.java.basic.datastructure.queue;


import java.util.concurrent.*;

/**
 * @author Tensai
 * 2019/3/29
 */
public class BlockingQueueTest {

	/**
	 * 测试方法
	 */
	private static void testBasket() {
		// 建立一个装苹果的篮子
		final Basket basket = new Basket();

		/**
		 *定义苹果生产者
		 */
		class Producer implements Runnable {
			@Override
			public void run() {
				for (; ; ) {
					// 生产苹果
					System.out.println("生产者准备生产苹果："
							+ System.currentTimeMillis());
					try {
						basket.produce();
					} catch (InterruptedException e) {
						e.printStackTrace();
					}
					System.out.println("生产者生产苹果完毕："
							+ System.currentTimeMillis());
					System.out.println("生产完后有苹果：" + basket.getAppleNumber() + "个");
					// 休眠300ms
					try {
						Thread.sleep(300);
					} catch (InterruptedException e) {
						e.printStackTrace();
					}
				}
			}
		}

		/**
		 * 定义苹果消费者
		 */
		class Consumer implements Runnable {
			
			@Override
			public void run() {
				try {
					while (true) {
						// 消费苹果
						System.out.println("消费者准备消费苹果："
								+ System.currentTimeMillis());
						basket.consume();
						System.out.println("消费者消费苹果完毕："
								+ System.currentTimeMillis());
						System.out.println("消费完后有苹果：" + basket.getAppleNumber() + "个");
						// 休眠1000ms
						Thread.sleep(1000);
					}
				} catch (InterruptedException ex) {
					ex.printStackTrace();
				}
			}
		}

		ExecutorService service = new ThreadPoolExecutor(1, 1, 0L,
				TimeUnit.MILLISECONDS, new LinkedBlockingQueue<>(), new ThreadPoolExecutor.AbortPolicy());
		Producer producer = new Producer();
		Consumer consumer = new Consumer();
		service.submit(producer);
		service.submit(consumer);
		// 程序运行10s后，所有任务停止
		try {
			Thread.sleep(10000);
		} catch (InterruptedException e) {
			e.printStackTrace();
		}
		service.shutdownNow();
	}

	public static void main(String[] args) {
		BlockingQueueTest.testBasket();
	}

	/**
	 * 定义装苹果的篮子
	 */
	public static class Basket {
		// 篮子，能够容纳3个苹果
		BlockingQueue<String> basket = new ArrayBlockingQueue<>(3);

		// 生产苹果，放入篮子
		void produce() throws InterruptedException {
			// put方法放入一个苹果，若basket满了，等到basket有位置
			basket.put("An apple");
		}

		// 消费苹果，从篮子中取走
		void consume() throws InterruptedException {
			// get方法取出一个苹果，若basket为空，等到basket有苹果为止
			basket.take();
		}

		int getAppleNumber() {
			return basket.size();
		}

	}
}

